import colorama
print(colorama.Fore.RED)
import torch
print(colorama.Fore.CYAN)
print(f"cuda is available:{torch.cuda.is_available()}")